#pragma once

#include <sys/wait.h>
#include <fcntl.h>
#include "../comm/util.hpp"
#include "../comm/log.hpp"

//只负责进行代码的编译

namespace ns_compiler
{
    //引入文件名拼接功能
    using namespace ns_util;
    using namespace ns_log;

    class Compiler{
    public:
        Compiler()
        {}
        ~Compiler()
        {}
        //返回值：编译成功返回true，编译失败返回false
        //参数：要编译的文件名（不带后缀）
        static bool Compile(const std::string& file_name)
        {
            pid_t pid = fork();
            if(pid < 0)
            {
                LOG(ERROR) << "内部错误，创建子进程失败" << "\n";
                return false;
            }

            else if(pid == 0)
            {
                umask(0);
                int _stderr = open(PathUtil::CompilerError(file_name).c_str(), O_WRONLY | O_CREAT, 0644);
                if(_stderr < 0){
                    LOG(WARNING) << "没有成功形成stderr文件" << "\n";
                    exit(1);
                }
                //重定向标准错误到_stderr
                dup2(_stderr, 2);

                //程序替换并不影响进程的文件描述符表
                //子进程:调用编译器完成代码的编译工作
                //g++ -o target src -std=c++11
                execlp("g++", "g++", "-o", PathUtil::Exe(file_name).c_str(),\
                PathUtil::Src(file_name).c_str(), "-D", "COMPILER_ONLINE", "-std=c++11", nullptr);
                LOG(ERROR) << "启动g++编译器失败，可能是参数错误" << "\n";
                //程序替换可能失败，直接退出
                exit(2);
            }
            else
            {
                waitpid(pid, nullptr, 0);
                //走到这里，意味着子进程退出了
                //判断是否编译成功,看看有没有对应的可执行程序
                if(FileUtil::IsFileExists(PathUtil::Exe(file_name)))
                {
                    LOG(INFO) << PathUtil::Src(file_name) << "编译成功" << "\n";
                    return true;
                }
            }
            LOG(ERROR) << "编译失败，没有形成可执行程序" << "\n";
            return false;
        }
    };
}